
// 导出页面为PDF格式
import html2Canvas from 'html2canvas'
import JsPDF from 'jspdf'
export default{
  install (Vue, options) {
    Vue.prototype.getPdf = function () {
      var title = this.htmlTitle
      html2Canvas(document.querySelector('#pdfDom'), {
        allowTaint: true
      }).then(function (canvas) {
         // eslint-disable-next-line new-cap
         const pdf = new JsPDF('p', 'mm', 'a4'); // A4纸，纵向
         
         const ctx = canvas.getContext('2d');
         const a4w = 206; const a4h = 277; // A4大小，210mm x 297mm，四边各保留210mm的边距，显示区域190x277
         const imgHeight = Math.floor(a4h * canvas.width / a4w); // 按A4显示比例换算一页图像的像素高度
         let renderedHeight = 0;
  
         while (renderedHeight < canvas.height) {
           const page = document.createElement('canvas');
           page.width = canvas.width;
           page.height = Math.min(imgHeight, canvas.height - renderedHeight);// 可能内容不足一页
  
           // 用getImageData剪裁指定区域，并画到前面创建的canvas对象中
           page.getContext('2d').putImageData(ctx.getImageData(0, renderedHeight, canvas.width, Math.min(imgHeight, canvas.height - renderedHeight)), 0, 0);
           console.log(page.toDataURL('image/jpeg', 1.0))
           pdf.addImage(page.toDataURL('image/jpeg', 1.0), 'JPEG', 1, 10, a4w, Math.min(a4h, a4w * page.height / page.width)); // 添加图像到页面，保留10mm边距
  
           renderedHeight += imgHeight;
           if (renderedHeight < canvas.height) {
             pdf.addPage();// 如果后面还有内容，添加一个空页
           }
           // 预览pdf(这里我用的是事件总线把canvas传递过去展示，达到模拟pdf预览的效果，有用但效果不是很好，有需要的可以自行修改)
           //this.$EventBus.$emit('open-pdf', canvas);
         }
          const totalPages = pdf.internal.getNumberOfPages() // 获取当前pdf的页数
          for (let i = 1; i <= totalPages; i++) {
            pdf.setPage(i) // 通过循环，跳转至相应的页进行设置
            // 后面可以进行相应的数据写入操作了，例如以下的添加水印操作
            pdf.setTextColor(150)
            console.log(pdf.internal.pageSize.width/2, pdf.internal.pageSize.height - 5)
            // pdf.text(X坐标, Y坐标, 文字内容)
            pdf.setFontSize(6);
            pdf.text(pdf.internal.pageSize.width/2, pdf.internal.pageSize.height - 5, String(i))
          }
         pdf.save(title + '.pdf')
      }
      )
    },
    Vue.prototype.getImg = function (name='capture') {
      html2Canvas(document.querySelector('#pdfDom'), {
        allowTaint: true
      }).then(function (canvas) {
        let img = new Image();
        img.src = canvas.toDataURL('image/png');
        // 或者可以直接下载图片
        let a = document.createElement('a');
        a.href = img.src;
        a.download = name+'.png';
        a.click();
      }
      )
    }
  }
}
